cd ../..

gpus=$1

# python joint_main.py \
#     --gpus ${gpus}, \
#     --opt_conf ./configs/opt_joint.yaml \
#     --exp_name joint/OOD_CRNN_w1.0_wMT_oRNN \
#     --feat_dim 256 \
#     --grad_compensate False \
#     --joint_itpl_exponent 0 \
#     --joint_itpl_warmup 80 \
#     --ood_weight 1.0

# Initialize CNN version

echo "unfreeze first 3 CNN layer"
python joint_main.py \
    --gpus ${gpus}, \
    --opt_conf ./configs/opt_joint.yaml \
    --exp_name joint/BYOLA_f3C_w1.0_woMT \
    --feat_dim 128 \
    --grad_compensate False \
    --joint_itpl_exponent 0 \
    --joint_itpl_warmup 80 \
    --ood_weight 1.0 \
    --batch_sizes "[0, 12, 12, 24, 256]" \
    --freeze_cnn_layer "[3, 4, 5, 6]" \
    --self_target cnn 